import gym
import nudt_env

env = gym.make('farm-tractor-v0') # 参数为环境的id

obs = env.reset()

for i in range(1000):
	action = agent(obs)
	obs, reward, done, info = env.step(action)
	if done:
		break